/*
 * Copyright (C) 2010-2011 ProjectSkyfire <http://www.projectskyfire.org/>
 * 
 * Copyright (C) 2011 ArkCORE <http://www.arkania.net/>
 * Copyright (C) 2008-2011 Trinity <http://www.trinitycore.org/>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

#include "TCSoap.h"
#include "soapH.h"
#include "soapStub.h"

void TCSoapRunnable::run() {
	struct soap soap;
	soap_init(&soap);
	soap_set_imode(&soap, SOAP_C_UTFSTRING);
	soap_set_omode(&soap, SOAP_C_UTFSTRING);

	// check every 3 seconds if world ended
	soap.accept_timeout = 3;
	soap.recv_timeout = 5;
	soap.send_timeout = 5;
	if (soap_bind(&soap, m_host.c_str(), m_port, 100) < 0) {
		sLog->outError("TCSoap: couldn't bind to %s:%d", m_host.c_str(),
				m_port);
		exit(-1);
	}

	sLog->outString("TCSoap: bound to http://%s:%d", m_host.c_str(), m_port);

	while (!World::IsStopped()) {
		if (!soap_valid_socket(soap_accept(&soap)))
			continue; // ran into an accept timeout

		sLog->outDebug(LOG_FILTER_NETWORKIO,
				"TCSoap: accepted connection from IP=%d.%d.%d.%d",
				(int) (soap.ip >> 24) & 0xFF, (int) (soap.ip >> 16) & 0xFF,
				(int) (soap.ip >> 8) & 0xFF, (int) soap.ip & 0xFF);
		struct soap* thread_soap = soap_copy(&soap); // make a safe copy

		ACE_Message_Block *mb = new ACE_Message_Block(sizeof(struct soap*));
		ACE_OS::memcpy(mb->wr_ptr(), &thread_soap, sizeof(struct soap*));
		process_message(mb);
	}

	soap_done(&soap);
}

void TCSoapRunnable::process_message(ACE_Message_Block *mb) {
	ACE_TRACE (ACE_TEXT ("SOAPWorkingThread::process_message"));

	struct soap* soap;
	ACE_OS::memcpy(&soap, mb->rd_ptr(), sizeof(struct soap*));
	mb->release();

	soap_serve(soap);
	soap_destroy(soap);
	// dealloc C++ data
	soap_end(soap); // dealloc data and clean up
	soap_done(soap); // detach soap struct
	free(soap);
}
/*
 Code used for generating stubs:

 int ns1__executeCommand(char* command, char** result);
 */
int ns1__executeCommand(soap* soap, char* command, char** result) {
	// security check
	if (!soap->userid || !soap->passwd) {
		sLog->outDebug(LOG_FILTER_NETWORKIO,
				"TCSoap: Client didn't provide login information");
		return 401;
	}

	uint32 accountId = sAccountMgr->GetId(soap->userid);
	if (!accountId) {
		sLog->outDebug(LOG_FILTER_NETWORKIO,
				"TCSoap: Client used invalid username '%s'", soap->userid);
		return 401;
	}

	if (!sAccountMgr->CheckPassword(accountId, soap->passwd)) {
		sLog->outDebug(LOG_FILTER_NETWORKIO,
				"TCSoap: invalid password for account '%s'", soap->userid);
		return 401;
	}

	if (sAccountMgr->GetSecurity(accountId) < SEC_ADMINISTRATOR) {
		sLog->outDebug(LOG_FILTER_NETWORKIO, "TCSoap: %s's gmlevel is too low",
				soap->userid);
		return 403;
	}

	if (!command || !*command)
		return soap_sender_fault(soap, "Command mustn't be empty",
				"The supplied command was an empty string");

	sLog->outDebug(LOG_FILTER_NETWORKIO, "TCSoap: got command '%s'", command);
	SOAPCommand connection;

	// commands are executed in the world thread. We have to wait for them to be completed
	{
		// CliCommandHolder will be deleted from world, accessing after queueing is NOT save
		CliCommandHolder* cmd = new CliCommandHolder(&connection, command,
				&SOAPCommand::print, &SOAPCommand::commandFinished);
		sWorld->QueueCliCommand(cmd);
	}

	// wait for callback to complete command

	int acc = connection.pendingCommands.acquire();
	if (acc) {
		sLog->outError(
				"TCSoap: Error while acquiring lock, acc = %i, errno = %u", acc,
				errno);
	}

	// alright, command finished

	char* printBuffer = soap_strdup(soap, connection.m_printBuffer.c_str());
	if (connection.hasCommandSucceeded()) {
		*result = printBuffer;
		return SOAP_OK;
	} else
		return soap_sender_fault(soap, printBuffer, printBuffer);
}

void SOAPCommand::commandFinished(void* soapconnection, bool success) {
	SOAPCommand* con = (SOAPCommand*) soapconnection;
	con->setCommandSuccess(success);
	con->pendingCommands.release();
}

////////////////////////////////////////////////////////////////////////////////
//
//  Namespace Definition Table
//
////////////////////////////////////////////////////////////////////////////////

struct Namespace namespaces[] = { { "SOAP-ENV",
		"http://schemas.xmlsoap.org/soap/envelope/", NULL, NULL }, // must be first
		{ "SOAP-ENC", "http://schemas.xmlsoap.org/soap/encoding/", NULL, NULL }, // must be second
		{ "xsi", "http://www.w3.org/1999/XMLSchema-instance",
				"http://www.w3.org/*/XMLSchema-instance", NULL }, { "xsd",
				"http://www.w3.org/1999/XMLSchema",
				"http://www.w3.org/*/XMLSchema", NULL }, { "ns1", "urn:TC",
				NULL, NULL }, // "ns1" namespace prefix
		{ NULL, NULL, NULL, NULL } };
